package com.itheima.security.springboot.controller;

import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.security.core.userdetails.UserDetails;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

/**
 * <p>
 * LoginController
 * </p>
 *
 * @author flyduck
 * @since 2023/11/18
 */
@RestController
public class LoginController {

    @RequestMapping(value = "/login-success", produces = {"text/plain;charset=utf8"})
    public String loginSuccess() {
        //提示具体的用户登陆成功
        return getUsername() + "登录成功";
    }

    /**
     * 测试资源1* @return
     */
    @PreAuthorize("hasAuthority('p1')")//拥有p1权限才可以访问
//    @PreAuthorize("hasAnyAuthority('p1','p3')")
    @GetMapping(value = "/r/r1", produces = {"text/plain;charset=utf8"})
    public String r1() {
        return getUsername() + " 访问资源1";
    }

    /**
     * 测试资源2* @return
     */
    @PreAuthorize("hasAuthority('p2')")
    @GetMapping(value = "/r/r2", produces = {"text/plain;charset=utf8"})
    public String r2() {
        return getUsername() + " 访问资源2";
    }

    //获取当前登陆用户的信息
    private String getUsername(){
        String username = "";
        //当前登录用户的身份
        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
        //用户身份
        Object principal = authentication.getPrincipal();

        if (principal == null) {
            username = "匿名";
        }

        if (principal instanceof UserDetails) {
            UserDetails userDetails = (UserDetails) principal;
            username = userDetails.getUsername();
        }else {
            username = principal.toString();
        }
        return username;
    }
}
